import unittest

from wcplus import python_lines
class Test_basic_function(unittest.TestCase):

    def test_py_lines_count(self):
        d, k, z = python_lines("testplus.py")
        self.assertEqual(d, 4)
        self.assertEqual(k, 5)
        self.assertEqual(z, 3)

if __name__ == '__main__':
    unittest.main()